package com.sbtr.business.importmodel.service;

import com.baomidou.mybatisplus.extension.service.IService;
import com.sbtr.business.importmodel.entity.ImportModelTask;
import com.sbtr.common.PageDto;
import com.sbtr.common.Response;
import com.sbtr.open.importmodel.dto.ImportModelTaskDto;
import com.sbtr.open.importmodel.dto.ImportModelTaskSearchDto;
import com.sbtr.open.publishtask.dto.PredictDataMultiNewDto;
import com.sbtr.open.publishtask.dto.PublishPredictTaskDto;
import com.sbtr.open.publishtask.dto.PublishTaskDataReqDto;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.util.List;

/**
 * <p>
 * 训练模型表 服务类
 * </p>
 *
 * @author yuyh2

 */
public interface IImportModelTaskService extends IService<ImportModelTask> {
    public PageDto<ImportModelTaskDto> search(ImportModelTaskSearchDto pro);

    public ImportModelTaskDto getById(String id);

    public ImportModelTaskDto save(ImportModelTaskDto dto);

    public Boolean delete(int id);

    public int batchDelete(List<Long> idList) ;

    public ImportModelTaskDto upload(
            Long id,//任务id
            MultipartFile file);

    /**
     * 发布外部模型:
     * 发送请求到python端
     * @param req
     * @return
     */
    public Response<String> publishExternalModel(PublishTaskDataReqDto req);

    /**
     * 下线外部模型:
     * 发送请求到python端
     * @param req
     * @return
     */
    public Response<Boolean> offlineExternalModel(PublishTaskDataReqDto req);


}
